package com.tgy.赫夫曼树;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;

/***
 * @ClassName: HuffmanTree
 * @Description: 赫夫曼树
 * @Auther: tanggy3
 * @Date: 10:10 2023/2/15
 * @version : V1.0
 */
public class HuffmanTree {
    public static void main(String[] args) {
        Node node1 = new Node(-8);
        Node node2 = new Node(5);
        Node node3 = new Node(-1);

//        int i = node1.compareTo(node2);
//        System.out.println(i);
//        ArrayList<Node> objects = new ArrayList<>();
//        objects.add(node1);
//        objects.add(node2);
//        objects.add(node3);
//        objects.forEach(System.out::println);
//        Collections.sort(objects);
//        objects.forEach(System.out::println);

        int arr[] = {13, 7, 8, 3, 29, 6, 1};
        Node root = createHuffmanTree(arr);

        System.out.println(root);
        preOrder(root);
    }


    // 创建赫夫曼树的方法

    /**
     * @param arr 需要创建成哈夫曼树的数组
     * @return 创建好后的赫夫曼树的root结点
     * 第一步为了操作方便
     * 1. 遍历 arr 数组
     * 2. 将arr的每个元素构成成一个Node
     * 3. 将Node 放入到ArrayList中
     */
    public static Node createHuffmanTree(int[] arr) {
        List<Node> nodes = new ArrayList<Node>();
        for (int value : arr) {
            nodes.add(new Node(value));
        }
        //对数组进行排序
        //Collections.sort(nodes);

        while (nodes.size() > 1) {
            System.out.println("nodes =" + nodes);
            //排序 从小到大
            Collections.sort(nodes, new Comparator<Node>() {
                @Override
                public int compare(Node o1, Node o2) {
                    return o1.getValue() > o2.getValue() ? 1 : -1;
                }
            });


            //取出根节点权值最小的两颗二叉树
            //(1) 取出权值最小的结点（二叉树）
            Node leftNode = nodes.get(0);
            //(2) 取出权值第二小的结点（二叉树）
            Node rightNode = nodes.get(1);
            //(3)构建一颗新的二叉树
            Node parent = new Node(leftNode.getValue() + rightNode.getValue());
            parent.setLeft(leftNode);
            parent.setRight(rightNode);

            //(4)从ArrayList删除处理过的二叉树
            nodes.remove(leftNode);
            nodes.remove(rightNode);
            //(5)将parent加入到nodes
            nodes.add(parent);
        }

        //返回哈夫曼树的root结点
        return nodes.get(0);
    }

    /**
     * 前序遍历
     * @param node
     */
    public static void preOrder(Node node){
        if(node == null){
            System.out.println("是空树，不能遍历~~");
        }else {
            System.out.println(node);
            if (node.getLeft()!=null){
                preOrder(node.getLeft());
            }
            if (node.getRight()!=null){
                preOrder(node.getRight());
            }
        }
    }
}
